package com.exercise.widgets.viewpager2

import android.content.Context
import android.util.AttributeSet
import android.view.MotionEvent
import android.view.View
import android.view.ViewConfiguration
import android.widget.FrameLayout
import androidx.viewpager2.widget.ViewPager2
import androidx.viewpager2.widget.ViewPager2.ORIENTATION_HORIZONTAL
import kotlin.math.absoluteValue
import kotlin.math.sign

/**
 * 处理ViewPager嵌套ViewPager或RecyclerView的冲突问题
 */
class NestScrollableHost : FrameLayout {

    constructor(context: Context) : super(context)
    constructor(context: Context, attrs: AttributeSet) : super(context, attrs)

    private var touchSlop = 0
    private var initialX = 0f
    private var initialY = 0f

    private val parentViewPager: ViewPager2?
        get() {
            var v = parent as? View
            while (v != null && v !is ViewPager2) {
                v = v.parent as? View
            }
            return v as? ViewPager2
        }
    private val child: View? get() = if (childCount > 0) getChildAt(0) else null

    init {
        touchSlop = ViewConfiguration.get(context).scaledTouchSlop
    }

    private fun canChildScroll(orientation: Int, delta: Float): Boolean {
        val direction = -delta.sign.toInt()
        return when (orientation) {
            0 -> child?.canScrollHorizontally(direction) ?: false
            1 -> child?.canScrollVertically(direction) ?: false
            else -> throw IllegalArgumentException()
        }
    }

    override fun onInterceptTouchEvent(ev: MotionEvent): Boolean {
        handleInterceptTouchEvent(ev)
        return super.onInterceptTouchEvent(ev)
    }

    private fun handleInterceptTouchEvent(e: MotionEvent) {
        // 获取外层ViewPager2的滑动方向
        val orientation = parentViewPager?.orientation ?: return
        // 判断嵌套的ViewPager和外层ViewPager的滑动方向是否相同，如果不相同，说明不会产生滑动冲突，直接return即可
        if (!canChildScroll(orientation, -1f) && !canChildScroll(orientation, 1f)) {
            return
        }

        // 嵌套的ViewPager2和外层的ViewPager2滑动方向相同，需要处理冲突
        // DOWN事件时记录初始坐标，
        if (e.action == MotionEvent.ACTION_DOWN) {
            initialX = e.x
            initialY = e.y
            parent.requestDisallowInterceptTouchEvent(true)
        } else if (e.action == MotionEvent.ACTION_MOVE) {// MOVE事件时，需要处理冲突

            // 计算x，y轴的滑动距离
            val dx = e.x - initialX
            val dy = e.y - initialY

            val isVpHorizontal = orientation == ORIENTATION_HORIZONTAL

            val scaledDx = dx.absoluteValue * if (isVpHorizontal) .5f else 1f
            val scaledDy = dy.absoluteValue * if (isVpHorizontal) 1f else .5f
            if (scaledDx > touchSlop || scaledDy > touchSlop) {
                // isVpHorizontal表示外层ViewPager水平滚动，scaledDy > scaledDx表示嵌套的内层ViewPager2垂直滚动
                if (isVpHorizontal == (scaledDy > scaledDx)) {
                    // 这里表示的是外层ViewPager2和内层ViewPager2滚动方向不一致，此时，不需要拦截事件
                    parent.requestDisallowInterceptTouchEvent(false)
                } else {
                    //父子ViewPager滑动方向一致

                    // 此时，如果子ViewPager也可以滑动，则拦截事件交给子ViewPager处理
                    if (canChildScroll(orientation, if (isVpHorizontal) dx else dy)) {
                        parent.requestDisallowInterceptTouchEvent(true)
                    } else {
                        // 子ViewPager不能滑动（比如只有一个View），则不需要拦截处理
                        parent.requestDisallowInterceptTouchEvent(false)
                    }
                }
            }
        }
    }

}